from transformers import BertTokenizer, BertForSequenceClassification, pipeline

# 设置具体包含 config.json的目录
model_dir = r"D:\AI\model\bert-base-chinese"  #这是一个文本分类模型

# 加载模型和分词器
model = BertForSequenceClassification.from_pretrained(model_dir)
tokenizer = BertTokenizer.from_pretrained(model_dir)

# 使用加载的模型和分词器创建生成文本的pipline
classifier = pipeline("text-classification", model=model, tokenizer=tokenizer, device=-1)  # -1表示CPU

# 进行分类
result = classifier("你好，我是一款语言模型，请给我一个评价")

print(result)

print(model)
